int32 episode_number
float32 episode_reward
